import requests  
  
# 设置你的Flask API的URL  
api_url = 'http://127.0.0.1:8000/api/images/generate'  # 注意：这里假设你的Flask应用运行在本地8000端口  
  
# 准备POST请求的数据  
data = {  
    'prompt': 'A cinematic shot of beautiful Korean woman wearing sun-dress.',
    'negative_prompt': 'ugly, blurry, low_res.',
    # 'prompt': 'A cinematic shot of a woman wearing an intricate italian priest robe.',
    # 'image': None,  
    'width': 512,  
    'height': 512,
    'guidance_scale': 0.0,
    'num_inference_steps': 2
}  
  
# 发送POST请求  
response = requests.post(api_url, json=data)  
  
# 检查响应状态码  
if response.status_code == 200:  
    # 如果API返回的是图像数据（假设是字节流），则直接保存或处理这些数据  
    # 这里我们只是简单地将其保存到文件中  
    with open('generated_image_test.jpg', 'wb') as f:  
        f.write(response.content)  
    print("图像已保存为 generated_image.jpg")  
else:  
    # 如果响应状态码不是200，则打印错误信息  
    print(f"请求失败，状态码：{response.status_code}，响应内容：{response.text}")